package org.example.dynamic_planning;

import java.util.Arrays;

public class sell_stocks_includes_the_freeze_period {
    public static void main(String[] args) {
        //买卖股票的最佳时机含冷冻期

        //给定一个整数数组prices，其中第  prices[i] 表示第 i 天的股票价格 。​
        //设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
        //卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
        //注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
        int[] prices = {1,4,2};
        int i = maxProfit(prices);
        System.out.println(i);
    }

    public static int maxProfit(int[] prices) {
        int[] dp = new int[4];
        dp[0] = -prices[0];
        for (int i = 1; i < prices.length; i++) {
            int temp0 = dp[0];
            int temp2 = dp[2];
            dp[0] = Math.max(Math.max(dp[0],dp[3]-prices[i]),dp[1]-prices[i]);
            dp[1] = Math.max(dp[1],dp[3]);
            dp[2] = Math.max(temp2,temp0+prices[i]);
            dp[3] = temp2;
            System.out.println(Arrays.toString(dp));
        }
        return Math.max(Math.max(dp[0],dp[1]),dp[2]);
    }
}
